import torchvision.datasets.mnist as mnist
